from dataloader import get_data
import torch
# train
DEVICE = torch.device("cuda" if torch.cuda.is_available() else 'cpu')
BATCH_SIZE = 32
DATA_PATH = 'data/tang.npz'
LEARNING_RATE=1e-2
EPOCHES=10

# model
EMB_DIM = 128
IN_HIDEN1=256
IN_HIDEN2=512
IN_LSTM=256
LSTM_LAYER=2
OUT_LSTM=256
OUT_HIDEN1=512
OUT_HIDEN2=256

# inference
MAX_GEN_LEN = 200